MOBILE_V2 = 'mobilenet_v2'
RESNET_152 = 'resnet152'
RESNET_101 = 'resnet101'
RESNET_50 = 'resnet50'
RESNET_34 = 'resnet34'
RESNET_18 = 'resnet18'
VIT_B_16 = 'vit_b_16'
VIT_B_32 = 'vit_b_32'
VIT_L_16 = 'vit_l_16'
VIT_L_32 = 'vit_l_32'
EFF_NET_V2_S = 'eff_net_v2_s'
EFF_NET_V2_L = 'eff_net_v2_l'
BERT = 'bert'

RESNETS = [RESNET_152, RESNET_101, RESNET_50, RESNET_34, RESNET_18]
EFF_NETS = [EFF_NET_V2_S, EFF_NET_V2_L]
CONVOLUTION_MODELS = RESNETS + [MOBILE_V2] + EFF_NETS
TRANSFORMER_MODELS = [VIT_B_16, VIT_L_16, VIT_B_32, VIT_L_32]
VISION_MODEL_CHOICES = CONVOLUTION_MODELS + TRANSFORMER_MODELS
